import torch
from torch import nn


class AddOneLayer(nn.Module):
    def __init__(self, *args, **kwargs) -> None:
        """
        初始化类的实例。

        使用*args和**kwargs允许传递任意数量的位置和关键字参数。
        这些参数被传递给基类的构造函数，通过super()调用。

        参数:
            *args: 位置参数，可传递任意数量的参数。
            **kwargs: 关键字参数，可传递任意数量的关键字参数。
        """
        super().__init__(*args, **kwargs)


    def forward(self, input):
        """
        实现输入数据的简单前向传播。

        该函数接收一个输入参数，并对其进行简单的加工操作，返回加工后的结果。

        参数:
            input: 输入的数据，可以是任意类型，本例中假设为数值类型。

        返回值:
            加工后的数据，即输入数据加上1。返回值的类型与输入数据相同。
        """
        # 将输入数据加1，实现简单的数值变换
        output = input + 1
        return output



addOneLayer = AddOneLayer()
x = torch.tensor(1.0)
output = addOneLayer(x)
print(output)
